package sys

import (
	"fmt"
	"github.com/spf13/viper"
	"os"
)

func InitConfig() {
	basePath := os.Getenv("IOT_PLAT")
	viper.SetConfigName("cfg")                       // 配置文件名称(无扩展名)
	viper.SetConfigType("yaml")                      // 指定配置类型为YAML
	viper.AddConfigPath(basePath + "/common/config") // 在当前目录查找配置文件
	// 添加以下代码读取配置
	if err := viper.ReadInConfig(); err != nil {
		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
			panic(fmt.Sprintf("配置文件未找到: %s/config/cfg.yaml", basePath))
		} else {
			panic(fmt.Sprintf("配置文件解析错误: %v", err))
		}
	}
}

func RootPath() string {
	return os.Getenv("IOT_PLAT")
}
